﻿//给定一个由 0 和 1 组成的矩阵 mat ，请输出一个大小相同的矩阵，
//其中每一个格子是 mat 中对应位置元素到最近的 0 的距离。
//
//两个相邻元素间的距离为 1 。
//
//输入：mat = [[0, 0, 0], [0, 1, 0], [0, 0, 0]]
//输出： [[0, 0, 0], [0, 1, 0], [0, 0, 0]]
//
//输入：mat = [[0, 0, 0], [0, 1, 0], [1, 1, 1]]
//输出： [[0, 0, 0], [0, 1, 0], [1, 2, 1]]
//
//提示：
//	m == mat.length
//	n == mat[i].length
//	1 <= m, n <= 10^4
//	1 <= m * n <= 10^4
//	mat[i][j] is either 0 or 1.
//	mat 中至少有一个 0

class Solution {
    int dx[4] = { 0, 0, 1, -1 };
    int dy[4] = { 1, -1, 0, 0 };

public:
    vector<vector<int>> updateMatrix(vector<vector<int>>& mat) {
        int m = mat.size(), n = mat[0].size();
        // dist[i][j] == -1表⽰：没有搜索过

        // dist[i][j] != -1表⽰：最短距离

        vector<vector<int>> dist(m, vector<int>(n, -1));
        queue<pair<int, int>> q;
        // 1.把所有的源点加⼊到队列中

        for (int i = 0; i < m; i++)
            for (int j = 0; j < n; j++)
                if (mat[i][j] == 0) {
                    q.push({ i, j });
                    dist[i][j] = 0;
                }

        // 2.⼀层⼀层的往外扩

        while (q.size()) {
            auto [a, b] = q.front();
            q.pop();
            for (int i = 0; i < 4; i++) {
                int x = a + dx[i], y = b + dy[i];
                if (x >= 0 && x < m && y >= 0 && y < n && dist[x][y] == -1) {
                    dist[x][y] = dist[a][b] + 1;
                    q.push({ x, y });
                }
            }
        }
        return dist;
    }
};